#!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Author  : lkm

"""
    多元线性回归模型：进行瘦肉量对眼肌面积、腿肉量、腰肉量的多元线性回归分析
"""
import pandas as pd
import statsmodels.formula.api as smf

def run_model(file, names, sep):
    """ 读入数据、跑模型 """
    data = pd.read_csv(file, names=names, sep=sep)
    data = data.iloc[:,1:]
    set_model(data)

def set_model(data):
    """ 建模 """
    # 第一步：计算皮尔逊相关系数
    corr = data.corr()
    print("查看皮尔逊相关系数", corr)
    # 第二步：多元线性回归模型建立
    model = "y~x1+x2+x3"
    lmodel = smf.ols(model, data).fit()
    print(lmodel.summary()) # 输出回归模型结果

if __name__ == '__main__':
    file = r'zhu.txt'
    names = ['order', 'y', 'x1', 'x2', 'x3']
    sep = '\t'
    run_model(file, names, sep)



